ComplexAbs

class mindradar.ComplexAbs

复数tensor逐元素求模。 计算表达式为:

\[out_i = \sqrt{a_i^2 + b_i^2},\]

其中 \(a\) 和 \(b\) 分别是复数的实部和虚部。

输入：

• input : 输入的复数tensor

输出：

tensor， 复数tensor的模

支持平台：

CPU

备注

输入数据类型为 complex64 时，输出数据类型为 float32。 数据类型为 complex128 时，输出数据类型为 float64。其他数据类型不支持。

示例代码：

import mindspore as ms
import numpy as np
import mindradar as mr

x = np.exp(3j * np.pi * np.arange(7) / 7).astype(np.complex64)
x_tensor = ms.Tensor(x)
net = mr.ComplexAbs()
out = net(x_tensor)
print(out)